"""
# -*- coding: utf-8 -*-
# @Time    : 2023/6/13 21:49
# @Author  : 王摇摆
# @FileName: Data.py
# @Software: PyCharm
# @Blog    ：https://blog.csdn.net/weixin_44943389?type=blog
"""

from sklearn.datasets import load_digits
from sklearn.model_selection import train_test_split
print('=============程序开始运行================')
digits = load_digits()
print('1. 数据集加载成功')

# 数据集切分
X_train, X_test, y_train, y_test = train_test_split(digits['data'], digits['target'], test_size=0.2)